Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Updating solvers module to use arraylias #281

Conversation

DanPuzzuoli
Copy link
Collaborator

@DanPuzzuoli DanPuzzuoli commented Oct 30, 2023

Summary

This PR updates all of the solvers folder to use arraylias, aside from the perturbative_solvers folder, which must be updated after the perturbation folder.

The new release notes describe the material change to the interface:

  • Solver.__init__ now takes arguments array_library and vectorized instead of evaluation_mode.
  • The logic for when Solver.solve decides to automatically JAX compile when simulating a list of schedules has been slightly modified.

Details and comments

All of the tests in the following command should be passing:

python -m unittest discover test.dynamics.models; python -m unittest discover test.dynamics.arraylias; python -m unittest discover test.dynamics.signals; python -m unittest discover test.dynamics.pulse; python -m unittest test.dynamics.solvers.test_diffrax_DOP5; python -m unittest test.dynamics.solvers.test_fixed_step_solvers; python -m unittest test.dynamics.solvers.test_jax_odeint; python -m unittest test.dynamics.solvers.test_lanczos; python -m unittest test.dynamics.solvers.test_solver_utils; python -m unittest test.dynamics.solvers.test_solver_functions; python -m unittest test.dynamics.solvers.test_solver_functions_interface; 

@DanPuzzuoli DanPuzzuoli changed the base branch from main to arraylias-integration-branch December 14, 2023 22:37
@DanPuzzuoli DanPuzzuoli force-pushed the solvers-arraylias branch 2 times, most recently from f65437d to 1c6c3ab Compare December 18, 2023 12:47
@DanPuzzuoli DanPuzzuoli marked this pull request as ready for review December 19, 2023 07:20
Copy link
Contributor

@to24toro to24toro left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thank you, @DanPuzzuoli .
I separated my PR comments from perturbation updatePR.
Plese see if when you have time.

qiskit_dynamics/solvers/solver_classes.py Outdated Show resolved Hide resolved
qiskit_dynamics/solvers/solver_utils.py Show resolved Hide resolved
test/dynamics/solvers/test_solver_classes.py Outdated Show resolved Hide resolved
@DanPuzzuoli
Copy link
Collaborator Author

https://github.com/Qiskit-Extensions/qiskit-dynamics/blob/88b2dfd357005f5241d0c059a814360059a91fae/qiskit_dynamics/models/__init__.py#L107

Also, need to fix it?

yes good catch - for now I don't want to fix documentation, I think it should all be fixed at the end when all of the changes have been made (to avoid having to make further unexpected changes).

Copy link
Contributor

@to24toro to24toro left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Sorry for late approval.

@DanPuzzuoli DanPuzzuoli merged commit 4993e00 into qiskit-community:arraylias-integration-branch Jan 19, 2024
2 checks passed
DanPuzzuoli added a commit that referenced this pull request Jan 19, 2024
DanPuzzuoli added a commit that referenced this pull request Jan 24, 2024
DanPuzzuoli added a commit that referenced this pull request Feb 5, 2024
DanPuzzuoli added a commit that referenced this pull request Feb 20, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants